Update test, add more screenshots