Shilpa Shetty Kundra spends quality time with son, Viaan