Update for generating png