Added heap class that I forgot last time