added code for first test