some styling up
fix