removed type annotations for python 2 compatibility