My solution for celtra task assignment