Problems 4, 5 and 6 done in Python