Add return statements