Removed python 2 from tests