model working again - just need to build class into it