Various cosmetic and cleanups
Implement new function for taking screenshots.