Add return statement