add main func for test
PEP8 and main