Added class selector, fixed Berserker heart bug