Fix main screen buttons