fixed a bug where I hardcoded one of the dimensions