Homework tree is working better

Signed-off-by: Daniel Pierce <paladin2350@gmail.com>