[Week1][Fixed the bug with the wrong invocation of count function (in fact that has no effect, but... this code is more correct)]