Added unit tests

Added 3 unit tests for each function.