Create Fixed Array Size Python Interview Coding