- Fix vertex shader
- Some improvements