Room to optimize code

Thanks, I fixed it.