Whitlisting file inputs

[radzo73]

Thanks for the ideas!
However when you mentioned 

Code:

filename,fileext = os.path.splitext(element) # replaces a line of yours
if fileext[1:].upper() != 'PNG':
  continue

What line would it replace?

nvm, got it working. Again, thanks!