Cut your environment so that you can collect the diamond in each stage. Stay away from lava, bugs, and prickly things that plants shoot.