Add relevant classes