Some more cleanup v2
Some more cleanup v2
Added exit funstionality