Addining initial unit tests