Cleanup and style for screenshots