Working

Adding to the controler class