Updating parsing to handle stages and adding initial styling