Axi Interview Question

1. Fixing the given code