Just a couple of modifications lol