Add removing classes function