added basic lightning, still very buggy