Fix typo and clean up

Fix typo in function name
Add display model
Delete tmp files created by program after using them
Add Input checking