updated it to work with pip