Add return statements, because, well...