added important if __name__ convention at the end