adding tests and trivial case