I decited to make a lil change... Enjoy