Removed unused import
Added heap comments