How To Put Input Into Array In Python