started to add tests