updated it slightly