Updates to app functionality by following class notes