Added class logic to album click