Add VC logic into mainscreen