Add more data + bug fixes