[GENERAL ISSUE] Improve code quality

[priyap286]

Describe the issue Improve code quality and code test coverage. Eg. https://github.com/aws-cloudformation/cloudformation-guard/pull/201#discussion_r730546785

Any examples N/A

NOTE: Please be sure that the templates, rules and logs you provide as part of your issue do not contain any sensitive information.

Operating System: N/A [eg, MacOS, Windows, Ubuntu, etc]

OS Version N/A [eg Catalina, 10, 18.04, etc]