Added the code, now need to get segues working