Styling feature

fixed styling issues