fix richard code round 2