Add some more convenience methods + make model forward calls type-safe

Add some more convenience methods + make model forward calls type-safe #63