Cleanup layer model

[erwinlambert]

• [x] Move sheet and sheet_utils together into layer model
• [x] Move plot code to separate function
• [x] Move functions out of class environment
• [x] Apply jit throughout

[erwinlambert]

Decided to go for the easy option and move away from the class environment.