Issue 4

fixing things.