Finished implementation but something is wrong