Added an include, variable definitions, and a print at the end