just simple code I wrote for exersice