Solution added to assignment