Oh well I found if you add both pieces of code it works perfect!

Thank you again.