add styling back