Adding some more to code