Updating histogram code for new output key class