Added more styling
Minor update