Added comment

Understanding the use of 'ones' function in dodging an unwanted loop.