fixing stuff
adding category functionality