Refactor code. ( Add BaseViewController )