small bug fix maybe
small funtionality changes