Added a proper control menu with reset function