working on more code