Yeah. Was little bit confused with some changes I made to the code but @Duck Nebuchadnezzar helped me to figure it out, so all is good.