Minor changes to code. Didn't affect functionality.