Update styles to match original look