Solution to first challenge in BitFOuntain Swift course