AOS would include filing for work authorization. You need to have valid EAD to work. The easiest way is to be able to stay on current by an extension. AOS is not immediate. Plan for problems. In order to minimize them and processing time, consider consulting an attorney for assistance.

