fixed a bug and added levels for testing