made the main a little bit nicer