Changed reference to old main class