adding work entity and some tests