Improve p2js physics