Add return to the methods that required it