Working solution for ass2