Removed puts statement at the end of the programme