Added some functionality and it doesn't work