followed to first test. But not working correctly