Cleanup

Made it look nicer.