removed comments and added test