Made minor modifications