Add viewController into callback block