Added a function to return a puzzle with only the first block pieces.
Added header guards and removed an extra include.