Added png files to repo! + some minor code change