writing initial unit tests