Cleaned up code. Children expands though very basic