just a quick mod for part 2