Making changes to output to align with assignment specifications
Problem 6 of assignment 1 solution
Clean up

Making the checks more concise

Signed-off-by: cybergrouch <cyber_grouch_lange@yahoo.ie>