Changed the code a littlebit.