Looks like you can pay for your state return ahead of time too. After you pay for federal, just click the "State" option in the left-hand navbar to begin the process of creating a State return. Then follow the steps above and the opportunity to pay for State will present itself.
