Atletico Madrid get their man back as Costa completes move