Update line numbers as appropriate

Removing __init__ moved some code around, update the line numbers as
appropriate so that the right sections of code get highlighted.