move from "__main__" to main function
minor update