Readded main class