Credit to for one part of the code