Make the code more maintainable

Despite the code around it.

Thanks for the advice, @jlee-r7!