Part4 almost working