Add more code comments