Add mainClass for assembly