started writing some unit tests