Update for installation via pip