minor code formatting and adding in enter function
minor code formatting and adding in enter function