small changes to look nicer