Changed parameters for smaller dataset