Codility lesson01 first two problems