polished code
polished up code