Small update for Saad. Don't lose context when page refresh. Still needs a little work. Maybe Saad will finish it. :)