user
You need to complete the following task:

Develop an algorithm to find the maximum number in a list.


assistant
def find_max(lst):
    max_num = 0
    for num in lst:
        if num > max_num:
            max_num = num
    return max_num