just made some code look nicer