Removed lemmata loop, updated if __name__ == '__main__' conditions