Added three unit tests