added test and models for work