Added diagonal line code (commented out)