chang to if __name__=='__main__'
fix tab size