done - applied the correct class