hw04 extension