Add main class and update dockerfile