Updated VB code