Added return to function