add option to leave pi unnormalized