Started trying to make unit tests