made two layer model a bit quicker