start to add tests