Return statements added to functions