Added the line plotting. Currently rough, need to update the get_estimate python code.