fixed the wrong answer issue
added the feedback
code refactor
added some graphics