Fix import error in py3
Use latest matplotlib version on travis