add solution to apartment code demo