Add Model back to package