Fix requirements and setup