Assignment 01 -- Expanded to make it interactive