Made some changes. Made it work.