Photoshop: "Compositions" in Photoshop

After Effect's compositions are an easy way to edit something that has already been edited, and the original can be manipulated, with the second composition being manipulated as well. I was wondering if something like this could be implemented into Photoshop. I find myself having to flatten the image in order to make changes to the whole image, and then having to undo everything to get the layers back to change something, then flatten and repeat. Could compositions work in Photoshop to eliminate this?