Add and remove classes based on window size