added first bit of code