Modified script to show better output