Added some class, and div tags, and a few comments