Added fix mistake functionality