Add and adjust include statements to make code more versatile.