I improved it a little. I cant tell if it is better or not. :p
Thanks @kqerpillar for the og game I tried to help (not very much).