Slight changes to load 3D model, and new class(Needs a bit more work and cleaning up)