HW submission. Working, needs formatting