Made script for testing, fixed generator