Re-add png function