Slight modifications to variable declarations in problem 001