Update: more styling changes