added missing stuff
added missing stuff