Made code look nicer