Agree you can leave the RE in the company or move to personal bank account once is taxed.  I would decide based on where you are earning the most interest.  Also agree keep in RE account or new one, either way is fine if you want to see it as a separate line on statements.  CPA will track the change in account, I usually do mine like this:

CR 3100 SH Investment - Money I put into company

DR 3200 SH Distribution - Money I took out of company

CR 3300 Retained Earnings -  Money I earned in company

Also agree tax person s/b tracking your Basis, which may differ slightly from above, but DR Debit must not exceed Credit balances.