Updated code

Task #66