Add return statment to save