cleaned up code based on reek