I was keeping the existing code.. 
line numbers 21 and 22