first working code
forgot to add the files