updated some more code