Fixing main-class