Added fist bits of code :)