Added more functionality to edit menu