Model loading is 99% working! Cleaned up model code as well.