Python drone code