Update tests and screenshots