Add functionality for presenting small wave, large wave, loading and loaded states