Made it nicer, hope it works