Actually works now

Small mistakes