Just made this a bit nicer.