Compressed model down to 4. Added code with Niclas