Adding proper unit tests