added first test and mocked prime number check