Adding class module unit tests