Moved imports into __main__