@beeselmane could have thought about this...

Changed main class