[TASK] Adds revert function to remove applied classes