Update that improves PNG and improves code function