improve on styling