Add missing flag.Parse()
Server review and refactoring