Add screenshots & TODO