Unit tests added

Unit test added to test connection with database