UI Change

Removed the Container Div and updated comments.