Changed something in the main again :)