Add function for running the whole program (so it can be easily tested).
Correct string output format (assignment says there should be no space on either side of the equals sign).