Fix & test flag behaviour
Test extending classes