added initial unit tests