Added correct styling back