Add possibility to take screenshots