My initial code
=====