Here is a revised version of the code with improved formatting and documentation: