Of course, but as this is an exercise, it is good to think the whole algorithm through in all its consequences and see what effects even small changes in the code have on the logic of the algorithm. Otherwise you are only
cargo-culting.
CountZero
A program should be light and agile, its subroutines connected like a string of pearls. The spirit and intent of the program should be retained throughout. There should be neither too little or too much, neither needless loops nor useless variables, neither lack of structure nor overwhelming rigidity." - The Tao of Programming, 4.1 - Geoffrey James