Add return statement for action