Fixing main example 01