updated it a little