working on bug

-Did work on line this it [Fixes #1]