Adding the edit mode to the main screen.