Adding retun statement