fixing a few little bugs
Merge pull request #1 from akh2103/fixing-the-lines

fixing a few little bugs