Started adding in more view controller code on the main VC