add screenshot functionality with a terrible UI