My homework for lesson six(unittest only of the first lesson)