added my initial code :)