minor code changes

for it to look nicer