Take User Input In Array Python