Revert to correct depth value
Simplified code
Added #include