Adding more functionality to editor