I fixed it now
This is my version which is pretty cool.