Cleaned up code/comments a little
Added custom AST parser test