neatened the code