added the fun part