Made it a little bit nicer