Added more features to the edit page