adding style, added div
removed unnecessary code