Added first version of code