Step 1 is labelled in the __init__ function. Should I perhaps add a comment like "View __init__ for step one" at the beginning of the train function?