Updated code

0.2