small changes to see it work out