Fleshed out assignment-0 TODO