Update 2

Added some more code