Add controller logic for session, seems to be working