Collimation of magnetically driven jets from accretion discs