fixed compiler errors and added model class