Added return statement to the program