Have been trying to get the lines to only be draw in straight lines
without much success. 

Currently testing different methods but I think there is problem with my
logic rather with the code