Add return statement docs