minor code update

Thanks to Lezalith and Gouvernathor from the LazCave.com Discord for a code review and suggestions on improving the code.

The changes do not affect functionality, they just improve the code.