add back in layers