correction button code cleaned up