Added unit3 code