Update in the main class!

I have now updated the class