simpler program
simpler code, if and else part was having an issue
easy program