Added initial weight solver. Still needs additional testing