Updated included class balance changes.

Signed-off-by: jedi95 <jedi95@gmail.com>