styling update

Styled the working version.