Your proposed modifications are somewhat vague. They seem to be good for full credit but please be more detailed so better feedback can be provided.