added the 2 last layers