Trying to add tests