Trying to write up unit tests