adding some unit tests would be good
adding some unit tests would be good