Updated prob3 to adhere to PEP8.