Change to make cocde button click focus back to $message after performing everything