Fixed the first task

Consider patter matching