cleanup - cleaning up the source code to make rubocop happy