Added temporary main class