Minor code update to freshen the VS code