Add ``after`` param to ``when`` method. #67

@txels This solves #67