Should I focus on after tax bucket?

Hi, was wondering if I should throttle down 401k contributions and focus on building up my brokerage account?

I would like to have some flexibility with barista/coast fire in my 50’s but haven’t thought too much into it yet.

Age: 41

401k: $550k

Roth IRA: $52k

Brokerage: $60k

HSA: 5k (just became available last year)

I’ve realized the before tax bucket is quite heavy so I’m thinking I should focus on the others.

Thanks!