More styling improvements
minor fix