I wrote the code as you said and I hope you like it.
https://repl.it/join/fjktbgeo-jadenahern