Working Single Class
Code modified for 2+ classes is working for one class now