Remove python 2 stuff