screenshot functionality + front-end fixes