Refactored the line drawing that we have nicely drawn lines and
restructured the code in general.