Add py3k support