Add return ok statement