Fix import style. Use standard script layout using __name__. Fix class names.