You can still work on OPT while taking class. It doesn't have to be necessarily after all course works are completed.