It would be nice if they could line up all the 'selected' states on one worksheet with columns going across the page.  It would free up a lot of 'real estate' (screen space) for returns that are multi-state.  I'd even go as far as including state withholding (non-resident withholding, state estimates, composite payments, PTET payments, etc.) on the individual returns at the bottom of those worksheets.