La till kommentarer och fixade så att programmet stänger av sig på ett korrekt vis