Added first bit of code