Add a few more rules to balanced, leave code messy