Store Input In Array Python Example