Added option to add input prompt and code output.
Fixed bugs.