Figured out the last problem