Adding small set of unit tests