Added some functions everthing should be working