Implemented basic unit tests