Refactor "EditMenu" class