- fix small bug in assigning + introduce logging
-  fix bug in model