Minor changes trying to understand the code