fixed bug, forgot to finish method