Fixed a minor bug in part 1