Small fix in parsing
Working on task #4